package com.beadwallet.alarm.util;

import com.alibaba.fastjson.JSONObject;
import com.beadwallet.alarm.entity.wechat.MyX509TrustManager;
import com.beadwallet.alarm.config.WeChatParames;
import com.beadwallet.alarm.entity.wechat.AccessToken;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import javax.net.ssl.TrustManager;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.ConnectException;
import java.net.URL;

/**
 * 微信工具类
 *
 * @author hyf
 * @date 2017/12/05
 */
public class WeChatUtil {

    private static Logger log = LoggerFactory.getLogger(WeChatUtil.class);

    /**
     * 发起https请求并获取结果
     *
     * @param requestUrl    请求地址
     * @param requestMethod 请求方式(GET,POST)
     * @param outputStr     提交的数据(Json)
     * @return JSONObject
     */
    public static JSONObject httpRequest(String requestUrl, String requestMethod, String outputStr) {
        JSONObject jsonObject = null;
        StringBuilder builder = new StringBuilder();
        try {
            // 创建SSLContext对象，并使用指定的信任管理器初始化
            TrustManager[] tm = {new MyX509TrustManager()};
            SSLContext sslContext = SSLContext.getInstance("SSL", "SunJSSE");
            sslContext.init(null, tm, new java.security.SecureRandom());
            // 从上述SSLContext对象中得到SSLSocketFactory对象
            SSLSocketFactory ssf = sslContext.getSocketFactory();

            URL url = new URL(requestUrl);
            HttpsURLConnection httpUrlConn = (HttpsURLConnection) url.openConnection();
            httpUrlConn.setSSLSocketFactory(ssf);

            httpUrlConn.setDoOutput(true);
            httpUrlConn.setDoInput(true);
            httpUrlConn.setUseCaches(false);
            // 设置请求方式（GET/POST）
            httpUrlConn.setRequestMethod(requestMethod);

            if (WeChatParames.GET.equalsIgnoreCase(requestMethod)) {
                httpUrlConn.connect();
            }
            // 当有数据需要提交时
            if (null != outputStr) {
                OutputStream outputStream = httpUrlConn.getOutputStream();
                // 注意编码格式，防止中文乱码
                outputStream.write(outputStr.getBytes("UTF-8"));
                outputStream.close();
            }

            // 将返回的输入流转换成字符串
            InputStream inputStream = httpUrlConn.getInputStream();
            InputStreamReader inputStreamReader = new InputStreamReader(inputStream, "UTF-8");
            BufferedReader bufferedReader = new BufferedReader(inputStreamReader);

            String str = null;
            while ((str = bufferedReader.readLine()) != null) {
                builder.append(str);
            }
            bufferedReader.close();
            inputStreamReader.close();
            // 释放资源
            inputStream.close();
            httpUrlConn.disconnect();
            jsonObject = JSONObject.parseObject(builder.toString());
        } catch (ConnectException ce) {
            log.error("微信服务器连接超时.");
        } catch (Exception e) {
            log.error("https 请求错误:{}", e);
        }
        return jsonObject;
    }

    /**
     * 获取AccessToken
     *
     * @param corpid     企业ID
     * @param corpsecret 应用凭证秘钥
     * @return AccessToken
     */

    public static AccessToken getAccessToken(String corpid, String corpsecret) {
        AccessToken accessToken = null;

        String requestUrl = WeChatParames.ACCESS_TOKEN_URL + "corpid=" + corpid + "&corpsecret=" + corpsecret;
        JSONObject jsonObject = httpRequest(requestUrl, "GET", null);
        //如果请求成功
        if (null != jsonObject) {
            accessToken = new AccessToken();
            accessToken.setToken(jsonObject.getString("access_token"));
            accessToken.setExpiresIn(jsonObject.getInteger("expires_in"));
        }
        return accessToken;
    }
}